#ifndef UNITLISTWIDGETITEM_H
#define UNITLISTWIDGETITEM_H

#include <QListWidgetItem>
#include "globaldata.h"

class UnitListWidgetItem : public QListWidgetItem
{
public:
    explicit UnitListWidgetItem(const QIcon &icon, const QString &text,
                             QListWidget *view = Q_NULLPTR, int type = Type);

    Unit getUnit() const { return m_unit; }
    void setUnit(const Unit &unit) { m_unit = unit; }

private:
    Unit m_unit;
};

#endif // UNITLISTWIDGETITEM_H
